#!/usr/bin/python
# -*- coding: UTF-8 -*-

import smtplib
from email.mime.text import MIMEText
from email.header import Header


def send_email():
    # 第三方 SMTP 服务
    mail_host = "smtp.tom.com"  # 设置服务器
    mail_user = "xc6@tom.com"  # 用户名
    mail_pass = "123456789s"  # 口令

    sender = 'xc6@tom.com'
    receivers = ['m15733897353@163.com']  # 接收邮件，可设置为你的QQ邮箱或者其他邮箱

    message = MIMEText('小戳哥哥提醒您，您关注的路程已经新开列车，请速去抢票！', 'plain', 'utf-8')
    message['From'] = Header("小戳抢票", 'utf-8')
    message['To'] = Header("小戳抢票用户", 'utf-8')

    subject = '小戳抢票：列车新开通知'
    message['Subject'] = Header(subject, 'utf-8')

    try:
        smtpObj = smtplib.SMTP()
        smtpObj.connect(mail_host, 25)  # 25 为 SMTP 端口号
        smtpObj.login(mail_user, mail_pass)
        smtpObj.sendmail(sender, receivers, message.as_string())
        print("邮件发送成功")
    except smtplib.SMTPException:
        print("Error: 无法发送邮件")


def send_error_email():
    # 第三方 SMTP 服务
    mail_host = "smtp.tom.com"  # 设置服务器
    mail_user = "xc6@tom.com"  # 用户名
    mail_pass = "123456789s"  # 口令

    sender = 'xc6@tom.com'
    receivers = ['m15733897353@163.com']  # 接收邮件，可设置为你的QQ邮箱或者其他邮箱

    message = MIMEText('小戳哥哥提醒您，您关注的路程暂未新开列车，已持续为您刷新中！！', 'plain', 'utf-8')
    message['From'] = Header("小戳抢票", 'utf-8')
    message['To'] = Header("小戳抢票用户", 'utf-8')

    subject = '小戳持续为您查询中...'
    message['Subject'] = Header(subject, 'utf-8')

    try:
        smtpObj = smtplib.SMTP()
        smtpObj.connect(mail_host, 25)  # 25 为 SMTP 端口号
        smtpObj.login(mail_user, mail_pass)
        smtpObj.sendmail(sender, receivers, message.as_string())
        print("邮件发送成功")
    except smtplib.SMTPException:
        print("Error: 无法发送邮件")


if __name__ == '__main__':
    send_email()
